package com.briup.es.config.auth;

import java.util.HashMap;
import java.util.Map;

/**
 * 用于存在用户id与其对应token映射的集合工具类，
 * 用于校验前端传入的token是否被冒用等情况
 */
public class TokenMap {
    private static final Map<Integer, String> MAP = new HashMap<>();

    public static void set(Integer userId, String token) {
        MAP.put(userId, token);
    }

    public static String get(Integer userId) {
        return MAP.get(userId);
    }

    public static void remove(Integer userId) {
        MAP.remove(userId);
    }
}
